from flask import Flask, render_template, request
from mar30.pcUtils import *

app = Flask(__name__)


@app.route("/")
def index():
    return render_template("index.html")


@app.route("/login", methods=['POST', 'GET'])
def login():
    username = request.form['username']  # 接受到输入的用户名
    print(username)
    userpwd = request.form['userpwd']  # 接受到输入的密码
    if username == 'admin':
        if userpwd == 'admin123':
            return render_template("pc.html")
        else:
            return "密码错误"
    else:
        return "当前用户不存在"


@app.route("/scrawl", methods=['POST', 'GET'])
def scrawl():
    name = request.form['name']
    url = request.form['targeturl']
    # 增加一个爬虫的业务
    s = pcScrawler()
    targetText = s.getTargetText(url, selector='p')
    return str(targetText)


if __name__ == "__main__":
    app.run()
